home *** CD-ROM | disk | FTP | other *** search
/ Language/OS - Multiplatform Resource Library / LANGUAGE OS.iso / scheme / sicp / stream.scm < prev    next >
Text File  |  1993-07-17  |  5KB  |  179 lines
  1. #| -*-Scheme-*-
  2.  
  3. $Header: stream.scm,v 1.1 90/09/10 18:12:21 GMT jinx Exp $
  4.  
  5. Copyright (c) 1987, 1988, 1989, 1990 Massachusetts Institute of Technology
  6.  
  7. This material was developed by the Scheme project at the Massachusetts
  8. Institute of Technology, Department of Electrical Engineering and
  9. Computer Science.  Permission to copy this software, to redistribute
  10. it, and to use it for any purpose is granted, subject to the following
  11. restrictions and understandings.
  12.  
  13. 1. Any copy made of this software must include this copyright notice
  14. in full.
  15.  
  16. 2. Users of this software agree to make their best efforts (a) to
  17. return to the MIT Scheme project any improvements or extensions that
  18. they make, so that these may be included in future releases; and (b)
  19. to inform MIT of noteworthy uses of this software.
  20.  
  21. 3. All materials developed as a consequence of the use of this
  22. software shall duly acknowledge such use, in accordance with the usual
  23. standards of acknowledging credit in academic research.
  24.  
  25. 4. MIT has made no warrantee or representation that the operation of
  26. this software will be error-free, and MIT is under no obligation to
  27. provide any services, by way of maintenance, update, or otherwise.
  28.  
  29. 5. In conjunction with products arising from the use of this material,
  30. there shall be no use of the name of the Massachusetts Institute of
  31. Technology nor of any adaptation thereof in any advertising,
  32. promotional, or sales literature without prior written consent from
  33. MIT in each case. |#
  34.  
  35. ;;;; Stream Utilities
  36.  
  37. (declare (usual-integrations))
  38.  
  39. ;;;; General Streams
  40.  
  41. (define (nth-stream n s)
  42.   (cond ((empty-stream? s)
  43.      (error "Empty stream -- NTH-STREAM" n))
  44.     ((= n 0)
  45.      (head s))
  46.     (else
  47.      (nth-stream (- n 1) (tail s)))))
  48.  
  49. (define (accumulate combiner initial-value stream)
  50.   (if (empty-stream? stream)
  51.       initial-value
  52.       (combiner (head stream)
  53.         (accumulate combiner
  54.                 initial-value
  55.                 (tail stream)))))
  56.  
  57. (define (filter pred stream)
  58.   (cond ((empty-stream? stream)
  59.      the-empty-stream)
  60.     ((pred (head stream))
  61.      (cons-stream (head stream)
  62.               (filter pred (tail stream))))
  63.     (else
  64.      (filter pred (tail stream)))))
  65.  
  66. (define (map-stream proc stream)
  67.   (if (empty-stream? stream)
  68.       the-empty-stream
  69.       (cons-stream (proc (head stream))
  70.            (map-stream proc (tail stream)))))
  71.  
  72. (define (map-stream-2 proc s1 s2)
  73.   (if (or (empty-stream? s1)
  74.       (empty-stream? s2))
  75.       the-empty-stream
  76.       (cons-stream (proc (head s1) (head s2))
  77.            (map-stream-2 proc (tail s1) (tail s2)))))
  78.  
  79. (define (append-streams s1 s2)
  80.   (if (empty-stream? s1)
  81.       s2
  82.       (cons-stream (head s1)
  83.            (append-streams (tail s1) s2))))
  84.  
  85. (define (enumerate-fringe tree)
  86.   (if (pair? tree)
  87.       (append-streams (enumerate-fringe (car tree))
  88.               (enumerate-fringe (cdr tree)))
  89.       (cons-stream tree the-empty-stream)))
  90.  
  91. ;;;; Numeric Streams
  92.  
  93. (define (add-streams s1 s2)
  94.   (cond ((empty-stream? s1) s2)
  95.     ((empty-stream? s2) s1)
  96.     (else
  97.      (cons-stream (+ (head s1) (head s2))
  98.               (add-streams (tail s1) (tail s2))))))
  99.  
  100. (define (scale-stream c s)
  101.   (map-stream (lambda (x) (* c x)) s))
  102.  
  103. (define (enumerate-interval n1 n2)
  104.   (if (> n1 n2)
  105.       the-empty-stream
  106.       (cons-stream n1 (enumerate-interval (1+ n1) n2))))
  107.  
  108. (define (integers-from n)
  109.   (cons-stream n (integers-from (1+ n))))
  110.  
  111. (define integers
  112.   (integers-from 1))
  113.  
  114. ;;;; Some Hairier Stuff
  115.  
  116. (define (merge s1 s2)
  117.   (cond ((empty-stream? s1) s2)
  118.     ((empty-stream? s2) s1)
  119.     (else
  120.      (let ((h1 (head s1))
  121.            (h2 (head s2)))
  122.        (cond ((< h1 h2)
  123.           (cons-stream h1
  124.                    (merge (tail s1)
  125.                       s2)))
  126.          ((> h1 h2)
  127.           (cons-stream h2
  128.                    (merge s1
  129.                       (tail s2))))
  130.          (else
  131.           (cons-stream h1
  132.                    (merge (tail s1)
  133.                       (tail s2)))))))))
  134.  
  135. ;;;; Printing
  136.  
  137. (define print-stream
  138.   (let ()
  139.     (define (iter s)
  140.       (if (empty-stream? s)
  141.       (write-string "}")
  142.       (begin (write-string " ")
  143.          (write (head s))
  144.          (iter (tail s)))))
  145.     (lambda (s)
  146.       (newline)
  147.       (write-string "{")
  148.       (if (empty-stream? s)
  149.       (write-string "}")
  150.       (begin (write (head s))
  151.          (iter (tail s)))))))
  152.  
  153. ;;;; Support for COLLECT
  154.  
  155. (define (flatmap f s)
  156.   (flatten (map-stream f s)))
  157.  
  158. (define (flatten stream)
  159.   (accumulate-delayed interleave-delayed
  160.               the-empty-stream
  161.               stream))
  162.  
  163. (define (accumulate-delayed combiner initial-value stream)
  164.   (if (empty-stream? stream)
  165.       initial-value
  166.       (combiner (head stream)
  167.         (delay (accumulate-delayed combiner
  168.                        initial-value
  169.                        (tail stream))))))
  170.  
  171. (define (interleave-delayed s1 delayed-s2)
  172.   (if (empty-stream? s1)
  173.       (force delayed-s2)
  174.       (cons-stream (head s1)
  175.            (interleave-delayed (force delayed-s2)
  176.                        (delay (tail s1))))))
  177.  
  178. (define ((spread-tuple procedure) tuple)
  179.   (apply procedure tuple))